#!/usr/bash
DATASET=$1
OUTPUT=$2
export ROOT=/public/home/lijg
export PATH=$PATH:$ROOT/anaconda3/bin
cd $ROOT/workspace/DM-GAN
source activate controlGAN
which python
cd code
python main.py --cfg ./cfg/{$DATASET}_DMGAN.yml --gpu 0 --data_dir ../data/$DATASET --manualSeed 1234 | tee OUTPUT